package com.webserver.http;

import java.io.*;
import java.net.Socket;
import java.nio.charset.StandardCharsets;
import java.util.HashMap;
import java.util.Map;
import java.util.Set;

import static com.webserver.http.HttpContext.*;

/**
 * 响应对象
 * 该类的每一个实例用于表示服务端发送给客户端的一个HTTP响应
 * 每个响应由三部分构成:
 * 状态行，响应头，响应正文
 */
public class HttpServletResponse {
    //状态行相关信息
    private int statusCode = 200;//状态代码
    private String statusReason = "OK";//状态描述

    //响应头相关信息
    //key:响应头的名字 value:响应头对应的值
    private Map<String,String> headers = new HashMap<>();

    //响应正文相关信息
    private File entity;//正文对应的实体文件

    private byte[] contentData;//以一组字节作为正文内容(通常为动态数据)
    /*
        java.io.ByteArrayOutputStream是一个低级流，其内部维护一个字节数组，通过这个流写出的字节
        数据都会转存到内部的字节数组上。
     */
    private ByteArrayOutputStream out;


    private Socket socket;
    public HttpServletResponse(Socket socket){
        this.socket = socket;
    }

    /**
     * 将当前响应对象内容按照标准的HTTP响应格式发送给客户端
     */
    public void response() throws IOException {
        //发送响应前的准备工作
        sendBefore();

        //3.1发送状态行
        sendStatusLine();

        //3.2发送响应头
        sendHeaders();

        //3.3发送响应正文
        sendContent();
    }

    /**
     * 发送响应前的准备工作
     */
    private void sendBefore(){
        //判断out是否为null，若不是null，说明处理时通过这个流写入了动态数据。
        if(out!=null){
            //将动态数据从ByteArrayOutputStream中获取回来
            contentData = out.toByteArray();
        }
        if(contentData!=null){//存在动态数据作为正文内容
            //根据该数组的长度设置响应头：Content-Length
            addHeader("Content-Length",contentData.length+"");
        }
    }


    private void sendStatusLine() throws IOException {
        String line = "HTTP/1.1"+" "+statusCode+" "+statusReason;
        println(line);
    }
    private void sendHeaders() throws IOException {
        //遍历headers将所有需要发送的响应头发送
        Set<Map.Entry<String,String>> entrySet = headers.entrySet();
        for(Map.Entry<String,String> e :entrySet){
            String key = e.getKey();
            String value = e.getValue();
            String line = key+": "+value;
            println(line);
        }

        //单独发送CRLF表示响应头部分发送完毕
        println("");
    }
    private void sendContent() throws IOException {
        OutputStream out = socket.getOutputStream();
        if(contentData!=null){//是否有动态数据作为正文内容
            out.write(contentData);
        }else if(entity!=null) {//正文文件存在则发送
            try (
                    FileInputStream fis = new FileInputStream(entity);
            ) {
                byte[] data = new byte[1024 * 10];//10kb
                int len;//记录每次实际读取到的字节量
                while ((len = fis.read(data)) != -1) {
                    out.write(data, 0, len);
                }
            }
        }
    }


    private void println(String line) throws IOException {
        OutputStream out = socket.getOutputStream();
        byte[] data = line.getBytes(StandardCharsets.ISO_8859_1);
        out.write(data);
        out.write(CR);//单独发送回车符
        out.write(LF);//单独发送换行符
    }

    public int getStatusCode() {
        return statusCode;
    }

    public void setStatusCode(int statusCode) {
        this.statusCode = statusCode;
    }

    public String getStatusReason() {
        return statusReason;
    }

    public void setStatusReason(String statusReason) {
        this.statusReason = statusReason;
    }

    public File getEntity() {
        return entity;
    }

    public void setEntity(File entity) {
        this.entity = entity;
        //获取文件名
        String fileName = entity.getName();//index.html
        //根据文件名获取后缀
        String ext = fileName.substring(fileName.lastIndexOf(".")+1);
        //根据后缀名作为key去mimeMapping这个Map中提取对应的value
        String mimeType = getMimeType(ext);
        //添加响应头Content-Type和Content-Length
        addHeader("Content-Type",mimeType);
        addHeader("Content-Length",entity.length()+"");
    }

    /**
     * 添加一个响应头
     * @param name
     * @param value
     */
    public void addHeader(String name,String value){
        this.headers.put(name,value);
    }

    /**
     * 获取一个字节输出流，通过该流写出的数据最终会作为正文发送给浏览器
     * @return
     */
    public OutputStream getOutputStream(){
        if(out == null){
            out = new ByteArrayOutputStream();
        }
        return out;
    }

    /**
     * 返回一个PrintWriter，通过该流写出的所有文本内容最终会作为响应正文发送给
     * 客户端
     * @return
     */
    public PrintWriter getWriter(){
        return new PrintWriter(
                new BufferedWriter(
                        new OutputStreamWriter(
                                getOutputStream(),StandardCharsets.UTF_8
                        )
                ),true
        );
    }

    public void setContentType(String value){
        addHeader("Content-Type",value);
    }

}
